tensor([[0.3400, 0.3400, 0.3400, 0.3400, 0.3400],
        [0.6600, 0.6600, 0.6600, 0.6600, 0.6600]], grad_fn=<RoundBackward1>)